package net.sunbeta.test.excel ;

import java.io.File ;
import java.io.FileInputStream ;
import java.io.FileOutputStream ;
import java.io.IOException ;
import java.io.OutputStream ;
import java.util.ArrayList ;
import java.util.List ;

import javax.servlet.http.HttpServletResponse ;

import org.apache.poi.hssf.usermodel.HSSFCell ;
import org.apache.poi.hssf.usermodel.HSSFRichTextString ;
import org.apache.poi.hssf.usermodel.HSSFRow ;
import org.apache.poi.hssf.usermodel.HSSFSheet ;
import org.apache.poi.hssf.usermodel.HSSFWorkbook ;
import org.apache.poi.poifs.filesystem.POIFSFileSystem ;
import org.apache.poi.ss.usermodel.Cell ;
import org.apache.poi.ss.usermodel.DateUtil ;
import org.apache.poi.ss.usermodel.Row ;
import org.apache.poi.ss.usermodel.Sheet ;
import org.apache.poi.ss.usermodel.Workbook ;
import org.joda.time.DateTime ;

/**
 * 封装对excel的操作，包括本地读写excel和流中输出excel<br/>
 * 关联jar poi-3.5-beta5-20090219.jar<br/>
 * 有参构造函数参数为excel的全路径<br/>
 * @author 沙琪玛
 *
 */
public class ExcelUtils {

    // excel文件路径
    private String path = "" ;

    /**
     * 无参构造函数 默认
     */
    public ExcelUtils() {

    }

    /**
     * 有参构造函数
     * @param path excel路径
     */
    public ExcelUtils(String path) {
        this.path = path ;
    }

    /**
     * 在磁盘生成一个含有内容的excel,路径为path属性
     * @param sheetName 导出的sheet名称
     * @param fieldName 列名数组
     * @param data 数据组
     * @throws IOException 
     */
    public void makeExcel(String sheetName, String[] fieldName, List<String[]> data) throws IOException {
        //在内存中生成工作薄
        HSSFWorkbook workbook = makeWorkBook(sheetName, fieldName, data) ;
        //截取文件夹路径
        String filePath = path.substring(0, path.lastIndexOf("\\")) ;
        // 如果路径不存在，创建路径
        File file = new File(filePath) ;
        //System.out.println(path+"-----------"+file.exists());
        if (!file.exists())
            file.mkdirs() ;
        FileOutputStream fileOut = new FileOutputStream(path) ;
        workbook.write(fileOut) ;
        fileOut.close() ;
    }

    /**
     * 在输出流中导出excel
     * @param excelName 导出的excel名称 包括扩展名
     * @param sheetName 导出的sheet名称
     * @param fieldName 列名数组
     * @param data 数据组
     * @param response response
     */
    public void makeStreamExcel(String excelName, String sheetName, String[] fieldName, List<String[]> data,
        HttpServletResponse response) {
        OutputStream os = null ;
        try {
            response.reset() ; // 清空输出流
            os = response.getOutputStream() ; // 取得输出流
            response.setHeader("Content-disposition", "attachment; filename=" + new String(excelName.getBytes(), "ISO-8859-1")) ; // 设定输出文件头
            response.setContentType("application/msexcel") ; // 定义输出类型
        } catch (IOException ex) {// 捕捉异常
            System.out.println("流操作错误:" + ex.getMessage()) ;
        }
        //在内存中生成工作薄
        HSSFWorkbook workbook = makeWorkBook(sheetName, fieldName, data) ;
        try {
            os.flush() ;
            workbook.write(os) ;
        } catch (IOException e) {
            e.printStackTrace() ;
            System.out.println("Output is closed") ;
        }
    }

    /**
     * 根据条件，生成工作薄对象到内存
     * @param sheetName 工作表对象名称
     * @param fieldName 首列列名称
     * @param data 数据
     * @return HSSFWorkbook
     */
    private HSSFWorkbook makeWorkBook(String sheetName, String[] fieldName, List<String[]> data) {
        // 产生工作薄对象
        HSSFWorkbook workbook = new HSSFWorkbook() ;
        // 产生工作表对象
        HSSFSheet sheet = workbook.createSheet() ;
        // 为了工作表能支持中文,设置字符集为UTF_16
        workbook.setSheetName(0, sheetName) ;
        // 产生一行
        HSSFRow row = sheet.createRow(0) ;
        // 产生单元格
        HSSFCell cell ;
        // 写入各个字段的名称
        for (int i = 0; i < fieldName.length; i++) {
            // 创建第一行各个字段名称的单元格
            cell = row.createCell((short) i) ;
            // 设置单元格内容为字符串型
            cell.setCellType(HSSFCell.CELL_TYPE_STRING) ;
            // 为了能在单元格中输入中文,设置字符集为UTF_16
            // cell.setEncoding(HSSFCell.ENCODING_UTF_16);
            // 给单元格内容赋值
            cell.setCellValue(new HSSFRichTextString(fieldName[i])) ;
        }
        // 写入各条记录,每条记录对应excel表中的一行
        for (int i = 0; i < data.size(); i++) {
            String[] tmp = data.get(i) ;
            // 生成一行
            row = sheet.createRow(i + 1) ;
            for (int j = 0; j < tmp.length; j++) {
                cell = row.createCell((short) j) ;
                //设置单元格字符类型为String
                cell.setCellType(HSSFCell.CELL_TYPE_STRING) ;
                cell.setCellValue(new HSSFRichTextString((tmp[j] == null) ? "" : tmp[j])) ;
            }
        }
        return workbook ;
    }

    public void write(int sheetOrder, int colum, int row, String content) throws Exception {
        Workbook workbook = new HSSFWorkbook(new POIFSFileSystem(new FileInputStream(path))) ;
        Sheet sheet = workbook.getSheetAt(sheetOrder) ;
        Row rows = sheet.createRow(row) ;
        Cell cell = rows.createCell(colum) ;
        cell.setCellValue(content) ;
        FileOutputStream fileOut = new FileOutputStream(path) ;
        workbook.write(fileOut) ;
        fileOut.close() ;

    }

    /**
     * 得到一个工作区最后一条记录的序号
     * @param sheetOrder 工作区序号
     * @return int
     * @throws IOException
     */
    public int getSheetLastRowNum(int sheetOrder) throws IOException {
        Workbook workbook = new HSSFWorkbook(new POIFSFileSystem(new FileInputStream(path))) ;
        Sheet sheet = workbook.getSheetAt(sheetOrder) ;
        return sheet.getLastRowNum() ;
    }

    public String read(int sheetOrder, int colum, int row) throws Exception {
        Workbook workbook = new HSSFWorkbook(new POIFSFileSystem(new FileInputStream(path))) ;
        Sheet sheet = workbook.getSheetAt(sheetOrder) ;
        Row rows = sheet.getRow(row) ;
        Cell cell = rows.getCell(colum) ;
        String content = cell.getStringCellValue() ;
        return content ;
    }

    /**
     * 根据path属性，在磁盘生成一个新的excel
     * @throws IOException
     */
    public void makeEmptyExcel() throws IOException {
        Workbook wb = new HSSFWorkbook() ;
        Sheet sheet = wb.createSheet("new sheet") ;
        //截取文件夹路径
        String filePath = path.substring(0, path.lastIndexOf("\\")) ;
        // 如果路径不存在，创建路径
        File file = new File(filePath) ;
        if (!file.exists())
            file.mkdirs() ;
        FileOutputStream fileOut = new FileOutputStream(filePath + "\\" + path.substring(path.lastIndexOf("\\") + 1)) ;
        wb.write(fileOut) ;
        fileOut.close() ;
    }

    /**
     * 根据工作区序号，读取该工作去下的所有记录，每一条记录是一个String[]<br/>
     * 注意如果单元格中的数据为数字将会被自动转换为字符串<br/>
     * 如果单元格中存在除数字，字符串以外的其他类型数据，将会产生错误
     * @param sheetOrder 工作区序号
     * @return
     * @throws IOException 
     * @throws  
     */
    public List<String[]> getDataFromSheet(int sheetOrder) throws IOException {
        Workbook workbook = new HSSFWorkbook(new POIFSFileSystem(new FileInputStream(path))) ;
        Sheet sheet = workbook.getSheetAt(sheetOrder) ;
        List<String[]> strs = new ArrayList<String[]>() ;
        //注意得到的行数是基于0的索引 遍历所有的行
        //System.out.println(sheet.getLastRowNum());
        for (int i = sheet.getFirstRowNum(); i <= sheet.getLastRowNum(); i++) {

            Row row = sheet.getRow(i) ;
            short minColIx = row.getFirstCellNum() ;
            short maxColIx = row.getLastCellNum() ;
            if (maxColIx == -1) {
                continue ;
            }
            String[] str = new String[maxColIx] ;
            int cellIndex = 0 ;
            for (short colIx = minColIx; colIx < maxColIx; colIx++) {
                Cell cell = row.getCell(colIx) ;
                if (cell == null) {
                    continue ;
                }

                str[cellIndex++] = getCellValue(cell) ;
            }
            strs.add(str) ;
        }
        return strs ;
    }

    private String getCellValue(Cell cell) {

        String cellValue = null ;
        switch (cell.getCellType()) {
            case Cell.CELL_TYPE_STRING:
                System.out.println(cell.getRichStringCellValue().getString()) ;
                cellValue = cell.getRichStringCellValue().getString() ;
                break ;
            case Cell.CELL_TYPE_NUMERIC:
                if (DateUtil.isCellDateFormatted(cell)) {
                    cellValue = new DateTime(cell.getDateCellValue().getTime()).toString("yyyyMMdd") ;
                } else {
                    cellValue = String.valueOf(cell.getNumericCellValue()) ;
                }
                break ;
            case Cell.CELL_TYPE_BOOLEAN:
                cellValue = String.valueOf(cell.getBooleanCellValue()) ;
                break ;
            case Cell.CELL_TYPE_FORMULA:
                cellValue = cell.getCellFormula() ;
                break ;
            default:
                cellValue = "" ;
        }
        return cellValue ;
    }

    public static void main(String[] args) throws Exception {
        ExcelUtils excelUtils = new ExcelUtils("D:\\temp\\FF_ECCS_File.xls") ;
        List<String[]> sheets = excelUtils.getDataFromSheet(0) ;

        List<String[]> ss = new ArrayList<String[]>() ;
        ss.add(new String[] {"你撒地方", "sdfds"}) ;
        ss.add(new String[] {"瓦尔", "撒地方"}) ;
        excelUtils.makeExcel("smsLog", new String[] {"色粉", "的是否"}, ss) ;
        List<String[]> strs = excelUtils.getDataFromSheet(0) ;
        String content = "Hello Worlds" ;
        excelUtils.write(2, 3, 1, content) ;
        String newContent = excelUtils.read(0, 1, 1) ;
        System.out.println(newContent) ;
        excelUtils.makeEmptyExcel() ;
    }
}
